#pragma once

#include <opencv2\opencv.hpp>

using namespace cv;

class GBoost
{
private:
	CvGBTrees boost;

public:

	void set_trainer(CvGBTrees _boost)
	{
		boost = _boost;
	}

	CvGBTrees get_trainer()
	{
		return boost;
	}

	void train(const Mat& data, const Mat& label, const CvGBTreesParams& params)
	{
		boost.train( data, CV_ROW_SAMPLE, label, Mat(), Mat(), Mat(), Mat(), params);
	}

	float predict(const Mat& sample)
	{
		return boost.predict(sample);
	}
};